%run set_theme.ipynb

import pandas as pd
import plotly.express as px
from plotly.offline import init_notebook_mode

init_notebook_mode()

import plotly.graph_objects as go

# Hieronder laad je de datasets in
df1 = pd.read_csv("StudentPerformanceFactors.csv")
df2 = pd.read_csv("gameandgrade.csv")
df3 = pd.read_csv("ResearchInformation3.csv")

# Gemiddelden berekenen
mean_scores = (
    df1.groupby(["Internet_Access", "Extracurricular_Activities"])["Exam_Score"]
    .mean()
    .reset_index()
    .round(1)
)

# Heatmap plotten
fig = px.density_heatmap(
    mean_scores,
    x="Internet_Access",
    y="Extracurricular_Activities",
    z="Exam_Score",
    text_auto=True,
    color_continuous_scale="Blues",  # Licht naar donker
    title="Gemiddelde Examenscore per Internettoegang en Activiteiten",
    labels={"Exam_Score": "Gemiddelde Examenscore"}
)

fig.update_layout(
    xaxis_title="Internettoegang",
    yaxis_title="Extracurriculaire Activiteiten",
    title_font_size=20,
    plot_bgcolor='white',
    coloraxis_colorbar=dict(title="Examenscore")
)

fig.show()